#include "AllInc.h"

/*
给定一个整数数组 nums 和一个目标值 target，请你在该数组中找出和为目标值的那 两个 整数，并返回他们的数组下标。

你可以假设每种输入只会对应一个答案。但是，你不能重复利用这个数组中同样的元素。

示例:

给定 nums = [2, 7, 11, 15], target = 9

因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
*/
// f1 两次遍历
// f2 利用一个set来加速
class Solution 
{
public:
    vector<int> twoSum(vector<int>& nums, int target) 
    {
		vector<int> result;
        map<int, int> records;
        size_t count = nums.size();
        for (size_t i = 0; i < count; i++)
        {
            int other = target - nums[i];
            const auto& r = records.find(other);
            if (r != records.end())
            {
                // find other
				result.push_back(r->second);
				result.push_back(i);

                return result;
            }
            else
            {
                records.insert(make_pair(nums[i], i));
            }
            
        }
        
        return result;
    }
};

//int main()
//{
//    Solution s;
//
//    vector<int> data{2, 7, 11, 15};
//    vector<int> res = s.twoSum(data, 9);
//
//    for (auto &&e : res)
//    {
//        cout << e << ",";
//    }
//
//    return 0;
//}